#ifndef KPEXTOPIMGRESAMPLEU16C1TOF32C1_H
#define KPEXTOPIMGRESAMPLEU16C1TOF32C1_H


#include "kpextVkContext.h"
#include "kpextBaseCompute.h"

namespace kpext {

/**
 * @brief 单通道32位单精度浮点数图像 感兴趣区域(ROI) 提取操作
 */
class OpImgResampleU16C1toF32C1 : public BaseCompute{
public:
    explicit OpImgResampleU16C1toF32C1(kpextVkContext * ctx);

    virtual ~OpImgResampleU16C1toF32C1() override;

    /**
     * @brief  重新构建计算管线及相关资源
     * @param cols  输入二维图像的列数
     * @param rows  输入二维图像的行数
     * @param inbuf  输入缓冲
     * @param idxbuf  1维重采样索引缓冲
     * @param outbuf  输出缓冲
     */
    virtual void rebuild(int cols,
                         int rows,
                         std::shared_ptr<BaseBuffer> inbuf,
                         std::shared_ptr<BaseBuffer> idxbuf,
                         std::shared_ptr<BaseBuffer> outbuf
                         );

protected:

    /**
     * @brief 计算着色器的work group size
     */
    std::array<uint32_t, 3> m_workGroupSize;

    /**
     * @brief 用于重采样的1维索引缓冲
     */
    BaseBuffer * m_idxBuf;

private:
    using BaseCompute::rebuild;

};  // class kpext::OpImgResampleU16C1toF32C1

} // namespace kpext

#endif // KPEXTOPIMGRESAMPLEU16C1TOF32C1_H
